import numpy as np
import torch
import random
import math
import torch.nn.functional as F
import matplotlib.pyplot as plt
import re
import torch.nn as nn
import copy
import warnings
import copy
from pathlib import Path

# a = torch.arange(4*5).reshape(4, 5).numpy()
b = np.arange(4*5).reshape(4, 5)
print(b)
print("*"*100)
print(b.cumsum(0))  # 从输出结果来看，是按照列累加
